from numpy import *
import torch

print(torch.cuda.is_available())
print(mat(random.rand(4,4)))